# 请提供train_set
import numpy as np
import torch
from pmdarima import auto_arima

stepwise_fit = auto_arima(train_set, start_p=1, start_q=1,
                          max_p=6, max_q=6, stepwise=True)
stepwise_fit.predict(n_periods=7)
